# coding=utf-8
import os

import pandas as pd  # 导入pandas库，用于读取Excel文件和处理数据

from excel.exceltools import exportExcel


def updateColumn(path):

    for root, dirs, files in os.walk(path):  #第一个为起始路径，第二个为起始路径下的文件夹，第三个是起始路径下的文件。
        for file in files:
            file_path=os.path.join(root,file)  #将路径名和文件名组合成一个完整路径
            # df = pd.read_excel(file_path,encoding="gbk") #excel转换成DataFrame
            df = pd.read_excel(file_path) #excel转换成DataFrame
            print(file_path)
            # 遍历Excel表格中的每一行，并将每一行插入到数据库中
            # for row in df.itertuples(index=False):  # 遍历DataFrame中的每一行
            #     tempList = list(row[4:19])
            #     tempList[0] = tempList[1] + tempList[0] + '_' + tempList[2]
            #     val = tuple(tempList)
                # val = row  # 插入的数据
            # todo
            df['SUB_NO'] = df.apply(lambda x: x['PGRID_NO'] + x['SUB_NO']+'_'+x['GRID_TYPE'], axis=1)


            # df.eval('SUB_NO1 = PGRID_NO + SUB_NO +"_"+ GRID_TYPE', inplace=True)
            # df.eval('SUB_NO1 = PGRID_NO + SUB_NO+ GRID_TYPE', inplace=True)
            # print(df)
            exportExcel(file_path, df)


if __name__ == '__main__':
    # 使用示例
    filename = r"F:\9.data\keti5\hunan\hunan_excel"

    updateColumn(filename)  # 调用函数，将Excel数据插入到MySQL数据库中
    print("完成")
